#pragma once
#include "compiler.hpp"
#include "runner.hpp"
#include <unistd.h>
#include <jsoncpp/json/json.h>
#include <vector>
#include "../comm/log.hpp"
#include "../comm/util.hpp"
namespace ns_compile_and_run
{
    using namespace ns_util;
    using namespace ns_log;
    using namespace ns_compiler;
    using namespace ns_runner;
    
    class CompileAndRun
    {    
    public:
        CompileAndRun()
        {}
        ~CompileAndRun()
        {}
        static void RemoveTempFile(const std::string& file_name) // 删除所有临时文件, 有才删(unlink)
        {
            std::vector<std::string> _vs;
            _vs.push_back(PathUtil::Compilerr(file_name));
            _vs.push_back(PathUtil::Src(file_name));
            _vs.push_back(PathUtil::Exe(file_name));
            _vs.push_back(PathUtil::Stderr(file_name));
            _vs.push_back(PathUtil::Stdin(file_name));
            _vs.push_back(PathUtil::Stdout(file_name));
            for(int i = 0; i < 6; i++)
            {
                if(FileUtil::IsFileExists(_vs[i]))
                    unlink(_vs[i].c_str());
            }
        }
        static std::string CodeToDesc(int status_code, const std::string& file_name)
        {
            std::string desc;
            switch(status_code)
            {
                case 0: desc = "运行成功";
                break;
                case -1: desc = "提交的代码是空的";
                break;
                case -2: desc = "未知错误";
                break;
                case -3: FileUtil::ReadFile(PathUtil::Compilerr(file_name), &desc, true); // 编译出错
                break;
                case 6: desc = "内存超过范围";
                break;
                case 8: desc = "浮点数溢出";
                break;
                case 11: desc = "段错误";
                break;
                case 24: desc = "CPU使用超时";
                break;
                default: desc = "未知错误";
                break;
            }
            return desc;
        }
        static void Start(const std::string& in_json, std::string* out_json)
        {
            Json::Value in_value;
            Json::Reader reader;
            // 反序列化
            reader.parse(in_json, in_value); 
            // 以 string 形式拿到 code 和 input
            std::string code = in_value["code"].asString();
            std::string input = in_value["input"].asString();
            // 以 int 形式拿到 cpu 和内存的限制
            int cpu_limit = in_value["cpu_limit"].asInt();
            int mem_limit = in_value["mem_limit"].asInt();

            int status_code = 0; // 状态码
            int run_result = 0; // 运行状态码
            /*******************
            * run_result < 0 内部错误
            * run_result = 0 运行成功
            * run_result > 0 运行过程种出错（有信号）
            ********************/
            std::string file_name; // 形成的唯一文件

            if(code.size() == 0) // 没有代码
            {
                status_code = -1; // 代码为空 
                goto END;
            }
            file_name = FileUtil::UniqueFileName();
            // 将code 的内容写入 file_name, 并形成源文件, 如果形成失败则进入下面步骤
            if(!FileUtil::WriteFile(PathUtil::Src(file_name), code))
            {
                status_code = -2; // 未知错误
                goto END;
            }
            if(!Compiler::Compile(file_name)) // 是否编译成功
            {
                status_code = -3; // 打开文件失败
                goto END;
            }
            run_result = Runner::Run(file_name, cpu_limit, mem_limit);

            if(run_result < 0)
            {
                status_code = -2; // 未知错误
            }
            else if(run_result > 0)
            {
                status_code = run_result;
            }
            else
            {
                status_code = 0;
            }
        END:
            Json::Value out_value;
            Json::StyledWriter writer;
            out_value["status"] = status_code;
            out_value["reason"] = CodeToDesc(status_code, file_name);
            if(status_code == 0)  
            {
                // 运行过程没问题
                std::string _stdout, _stderr;
                FileUtil::ReadFile(PathUtil::Stdout(file_name), &_stdout, true);
                FileUtil::ReadFile(PathUtil::Stderr(file_name), &_stderr, true);

                out_value["stdout"] = _stdout;
                out_value["stderr"] = _stderr;
            }

            (*out_json) = writer.write(out_value);
            // 删除临时文件 
            RemoveTempFile(file_name);
        }
    };
}